package com.yhn.geekanime.utils;

import cn.hutool.core.collection.CollUtil;
import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.support.ExcelTypeEnum;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.ObjectUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * Excel工具类，将excel转为csv文件
 *
 * @author yhnyhn123
 * @version 1.0
 * @description 将excel文档转化为csv格式的文件
 * @date 2024/2/27 16:26
 */
@Slf4j
public class ExcelUtils {

    /**
     * 将excel转为csv文件
     *
     * @param multipartFile 上传的excel文件
     * @return csv文件
     */
    public static String excelToCsv(MultipartFile multipartFile) {
        List<Map<Integer, String>> list = null;
        try {
            // 读取上传的Excel文件
            list = EasyExcel.read(multipartFile.getInputStream())
                    .excelType(ExcelTypeEnum.XLSX)
                    .sheet()
                    .headRowNumber(0)
                    .doReadSync();
        } catch (IOException e) {
            log.error("表格处理错误", e);
        }
        if (CollUtil.isEmpty(list)) {
            return "";
        }
        // 转换为 csv
        StringBuilder stringBuilder = new StringBuilder();
        // 读取表头
        // 获取第一个元素，即headerMap
        LinkedHashMap<Integer, String> headerMap = (LinkedHashMap) list.get(0);
        // 获取headerMap的值列表，并过滤掉空值
        List<String> headerList = headerMap.values().stream().filter(ObjectUtils::isNotEmpty).collect(Collectors.toList());
        // 将headerList中的元素以逗号分隔，并添加换行符，追加到stringBuilder中
        stringBuilder.append(StringUtils.join(headerList, ",")).append("\n");
        // 遍历剩余的元素，即dataMap
        for (int i = 1; i < list.size(); i++) {
            LinkedHashMap<Integer, String> dataMap = (LinkedHashMap) list.get(i);
            // 获取dataMap的值列表，并过滤掉空值
            List<String> dataList = dataMap.values().stream().filter(ObjectUtils::isNotEmpty).collect(Collectors.toList());
            // 将dataList中的元素以逗号分隔，并添加换行符，追加到stringBuilder中
            stringBuilder.append(StringUtils.join(dataList, ",")).append("\n");
        }

        return stringBuilder.toString();

    }
}
